#!/bin/bash
#SBATCH --job-name=sft_qwen_vl_25_3b_instruct
#SBATCH --partition=H100
#SBATCH --gres=gpu:4
#SBATCH --nodes=1
#SBATCH --cpus-per-task=32
#SBATCH --mem=128G
#SBATCH --time=24:00:00
#SBATCH --output=slurm-%j.out
#SBATCH --error=slurm-%j.err

# 加载环境
source /public/home/jianghaitenggroup/jianglab/.bashrc
conda activate torch251_py310_cu122

# 进入工作目录
cd /public/home/jianghaitenggroup/jianglab/WorkSpace/Qwen25_VL_FT/

# 准备数据
python ./prepare_data.py \
    --img_dir "/public/home/jianghaitenggroup/jianglab/WorkSpace/MASS-SS3_Wave_224_448_V2/" \
    --save_dir "./saved" \
    --assistant_dir "/public/home/jianghaitenggroup/jianglab/WorkSpace/MASS-SS3_Qwen2.5-VL-72B-Instruct/" \
    --system_prompt "./system_prompt.md" \
    --split_seed 42

# 启动训练
bash ./run_train.sh

# 启动验证集评估
bash ./val.sh

# 启动测试集评估
bash ./test.sh

